#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <pthread.h>
#include <unistd.h>
#include <errno.h>
#include <ctype.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <strings.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/wait.h>

# define SERV_PORT 9527

//多线程服务器
void sys_err(const char *str)
{
	perror(str);
	exit(1);
}

void *doWork(void *arg)//子线程要干的事
{
    int cnfd = (int) arg;//把收到的参数强制类型转换
	//所有线程共享lfd、cfd,所以不用香进程那样关闭lfd
	char buf[4096];
	int fret, i;

	while(1){
	    fret = read(cnfd,buf,sizeof(buf));
		if(fret == 0){
			printf("client %d closed\n", cnfd);
			break;//检测客户端是否断开，断开则跳到外面，关闭当前进程 
		}

		write(STDOUT_FILENO, buf ,fret);

		for(i=0; i<fret; i++){
			buf[i] = toupper(buf[i]);//对数据进行操作			
		}

		write(cnfd, buf, fret);
	}

	close(cnfd);
	return (void *)0; //这句作用等同于pthread_exit(0)
}

int main(int argc, char *argv[])
{
	int lfd = 0, cfd = 0; //lfd接收socket创建的文件描述符,cfd接受accept创建的与客户端通信的文件描述符
	pthread_t tid;//子线程id
	char clit_ip[1024];

	struct sockaddr_in	serv_addr, clit_addr;
	socklen_t	clit_addr_len;

	bzero(&serv_addr, sizeof(serv_addr));//将地址空间清零，即初始化，也可以用memset函数
	serv_addr.sin_family = AF_INET;
	serv_addr.sin_port = htons(SERV_PORT);//本地字节序转网络字节序
	serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);//任意可用ip地址

	lfd = socket(AF_INET,SOCK_STREAM,0);
	if(lfd == -1){
		sys_err("server socket create error");
	}

	int ret = bind(lfd,(struct sockaddr *)&serv_addr,sizeof(serv_addr));
	if(ret == -1){
		sys_err("server bind error");
	}

	ret = listen(lfd, 128);
	if(ret == -1){
		sys_err("server listen error");
	}
	
	//上面是监听socket

	clit_addr_len = sizeof(clit_addr);//传入只用求一次，放在循环里会反复求

	while(1){

		cfd = accept(lfd, (struct sockaddr *)&clit_addr, &clit_addr_len);
		if(cfd == -1){
			sys_err("serveraccept error");
		}
		printf("client ip:%s, port:%d\n", 
				inet_ntop(AF_INET, &clit_addr.sin_addr.s_addr, clit_ip, sizeof(clit_ip)),
				ntohs(clit_addr.sin_port));

		ret = pthread_create(&tid, NULL, doWork,(void *)cfd) ;//创建子线程
		if (ret != 0){
			sys_err("thread create error");
		}

		pthread_detach(tid);//线程分离，防止产生僵尸进程

	}

	return 0;
}

